# -*- coding:utf-8 -*-
__author__ = 'Kalyugin Pavel'

import os
from HW_PO.Pages.DetailedFilmPage import DetailedFilmPage


class AddFilmPage(object):
    def __init__(self, driver):
        self.driver = driver

    def add_film(self, name, year, name_of_photo):
        # Заполнение двух обязательных полей (название фильма и год)
        self.driver.find_element_by_name("name").send_keys(name)
        self.driver.find_element_by_name("year").send_keys(year)

        # Загрузка изображения фильма (Титаника)
        self.driver.find_element_by_xpath("//input[@type='file']").send_keys(os.getcwd() + "\\" + name_of_photo)

        # Кнопка сохранения фильма в базе данных (Save)
        self.driver.find_element_by_xpath("(//*[@id='content']//a)[1]").click()

        return DetailedFilmPage(self.driver)